#!/bin/bash
#SBATCH --job-name=ecg_sk
#SBATCH --partition=universe
#SBATCH --output=slurm/result_%j.txt
#SBATCH --error=slurm/error_%j.txt
#SBATCH --mem=70G
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=4
#SBATCH --gres=gpu:1
#SBATCH --time=10:00:00

echo "start time: $(date)"
echo "SLURM_JOBID="$SLURM_JOBID
echo "SLURM_JOB_NODELIST"=$SLURM_JOB_NODELIST
echo "SLURM_JOB_PARTITION"=$SLURM_JOB_PARTITION
echo "SLURM_NNODES"=$SLURM_NNODES
echo "SLURM_GPUS_ON_NODE"=$SLURM_GPUS_ON_NODE
echo "SLURM_SUBMIT_DIR"=$SLURM_SUBMIT_DIR

module load python/anaconda3
source activate mmcl

cd ECG-CMR-TIME-V2
export PYTHONPATH=$(pwd)

srun python3 train_evaluate.py